python main.py \
    --model opt \
    --model_args pretrained=facebook/opt-6.7b,model_cache_dir=weight/opt66b_checkpoints,tokenizer_cache_dir=weight/opt66b_tokenizer,mask_heads=1,head_importance_path=logs/head_importance/opt66b/1shot_piqa.pkl,head_percent_mask=20 \
    --tasks piqa \
    --output_path results/66b/piqa/1shot_piqa_percent.txt \
    --batch_size 2 \
    --num_fewshot 1